package com.cb2.algorithm.leetcode;

import java.util.HashMap;
import java.util.Map;

/**
 * <a href='https://leetcode.cn/problems/two-sum/description/'>两数之和(Two Sum)</a>
 * <p>给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target  的那 两个 整数，并返回它们的数组下标。</p>
 * <p>你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。</p>
 * <p>你可以按任意顺序返回答案。</p>
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1：
 *      输入：nums = [2,7,11,15], target = 9
 *      输出：[0,1]
 *      解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
 * 示例 2：
 *      输入：nums = [3,2,4], target = 6
 *      输出：[1,2]
 *
 * 示例 3：
 *      输入：nums = [3,3], target = 6
 *      输出：[0,1]
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 *     <ul>
 *      <li>2 <= nums.length <= 10^4</li>
 *      <li>-10^9 <= nums[i] <= 10^9</li>
 *      <li>-10^9 <= target <= 10^9</li>
 *      <li>只会存在一个有效答案</li>
 *     </ul>
 * </p>
 * <b>进阶：你可以想出一个时间复杂度小于 O(n^2) 的算法吗？</b>
 *
 * @author c2b
 * @since 2023/5/5 10:43
 */
public class LC0001TwoSum_S {

    static class Solution {
        public int[] twoSum(int[] nums, int target) {

            Map<Integer, Integer> value2IndexMap = new HashMap<>(nums.length - 1);
            for (int idx = 0; idx < nums.length; idx++) {
                if (value2IndexMap.containsKey(target - nums[idx])) {
                    return new int[]{idx, value2IndexMap.get(target - nums[idx])};
                }
                value2IndexMap.put(nums[idx], idx);
            }
            return new int[]{-1, -1};
        }

        // Map + 双指针
        public int[] twoSum2(int[] nums, int target) {
            // key数字中对于下标的值，value是下标
            Map<Integer, Integer> value2IndexMap = new HashMap<>();
            int leftIndex = 0;
            int rightIndex = nums.length - 1;
            while (leftIndex <= rightIndex) {
                if (value2IndexMap.containsKey(target - nums[leftIndex])) {
                    return new int[]{leftIndex, value2IndexMap.get(target - nums[leftIndex])};
                }
                value2IndexMap.put(nums[leftIndex], leftIndex++);
                if (value2IndexMap.containsKey(target - nums[rightIndex])) {
                    return new int[]{rightIndex, value2IndexMap.get(target - nums[rightIndex])};
                }
                value2IndexMap.put(nums[rightIndex], rightIndex--);
            }
            return new int[0];
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        for (int i : solution.twoSum(new int[]{1, 5, 2}, 10)) {
            System.out.println(i);
        }
    }
}